import re

def extract_data_label(pkl_item):
    text = pkl_item.prompt
    # 大类的企业:{industry_info}。\n将它分类到如下类别:
    item = re.search("请仔细阅读企业信息：(.*?)  \n接下来", text, re.DOTALL)
    info = item.group(1)
    label = pkl_item.outputs[0].text
    return info, label

def extract_text_from_prompt(prompt):
    # 大类的企业:{industry_info}。\n将它分类到如下类别:
    item = re.search("请仔细阅读企业信息：(.*?)  \n接下来", prompt, re.DOTALL)
    return item